package com.tian.common.util;

import java.util.Date;
import java.util.HashSet;
import java.util.Random;
import java.util.Set;

/**
 * {@code @description:} TODO
 *
 * @author tianwc 公众号：Java后端技术全栈
 * 在线刷题 1200+java面试题和1000+篇技术文章：<a href="https://woaijava.cc/">博客地址</a>
 * {@code @date:} 2024-10-21 20:11
 * {@code @version:} 1.0
 */
public class CouponCodeUtil {
    /**
     * 自定义进制(0,1没有加入,容易与o,l混淆)，数组顺序可进行调整增加反推难度，A用来补位因此此数组不包含A，共31个字符。
     */
    private static final char[] BASE = new char[]{'H', 'V', 'E', '8', 'S', '2', 'D', 'Z', 'X', '9', 'C', '7', 'P',
            '5', 'I', 'K', '3', 'M', 'J', 'U', 'F', 'R', '4', 'W', 'Y', 'L', 'T', 'N', '6', 'B', 'G', 'Q'};
    /**
     * A补位字符，不能与自定义重复
     */
    private static final char SUFFIX_CHAR = 'A';

    /**
     * 进制长度
     */
    private static final int BIN_LEN = BASE.length;
    /**
     * 生成邀请码最小长度
     */
    private static final int CODE_LEN = 6;

    public static String couponCode(Integer createUserId) {
        String idToCodes = DateUtils.format(new Date(), DateUtils.FORMAT_NO_SECOND) + "" + createUserId;
        Long num = Long.parseLong(idToCodes);
        char[] buf = new char[BIN_LEN];
        int charPos = BIN_LEN;
        // 当id除以数组长度结果大于0，则进行取模操作，并以取模的值作为数组的坐标获得对应的字符
        while (num / BIN_LEN > 0) {
            int index = (int) (num % BIN_LEN);
            buf[--charPos] = BASE[index];
            num /= BIN_LEN;
        }
        buf[--charPos] = BASE[(int) (num % BIN_LEN)];
        // 将字符数组转化为字符串
        String result = new String(buf, charPos, BIN_LEN - charPos);
        // 长度不足指定长度则随机补全
        int len = result.length();
        if (len < CODE_LEN) {
            StringBuilder sb = new StringBuilder();
            sb.append(SUFFIX_CHAR);
            Random random = new Random();
            for (int i = 0; i < CODE_LEN - len - 1; i++) {
                sb.append(BASE[random.nextInt(BIN_LEN)]);
            }
            result += sb.toString();
        }

        return result;
    }

    public static void main(String[] args) {
        Set<String> set = new HashSet<>();

        for (int i = 0; i < 100000; i++) {
            set.add(couponCode(i));
        }
        System.out.println(set.size());//100000
    }
}
